python code for avengers